import os
from openai import OpenAI

client = OpenAI(
    api_key="sk-ce243630991443b2a32a217c15f8128c",  # 如果您没有配置环境变量，请在此处用您的API Key进行替换
    base_url="https://dashscope.aliyuncs.com/compatible-mode/v1"  # 百炼服务的base_url
)

embedding = client.embeddings

completion = client.embeddings.create(
    model="text-embedding-v4",
    input='衣服的质量杠杠的，很漂亮，不枉我等了这么久啊，喜欢，以后还来这里买',
    dimensions=1024, # 指定向量维度（仅 text-embedding-v3及 text-embedding-v4支持该参数）
    encoding_format="float"
)

print(completion.model_dump_json())


from langchain_dashscope import DashScopeEmbeddings

# 初始化嵌入模型
embeddings = DashScopeEmbeddings(
    model="text-embedding-v4",  # 可选：text-embedding-v1, text-embedding-v2
    api_key="sk-ce243630991443b2a32a217c15f8128c",
)

# 嵌入文档
documents = ["量子力学是研究微观世界的物理学分支", "人工智能是模拟人类智能的技术"]
vectors = embeddings.embed_documents(documents)
print(f"向量维度: {len(vectors[0])}")

# 嵌入查询
query_vector = embeddings.embed_query("什么是量子纠缠")
print(f"查询向量维度: {len(query_vector)}")

